package com.arges.web.controller;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Scope;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import com.arges.web.common.AESUtils;
import com.arges.web.common.Constant;
import com.arges.web.domain.bean.DocBean;
import com.arges.web.service.DocManager;

/**
 * 登陆控制层
 * 
 * @author arges
 *
 */
@RestController
@Scope("prototype")
@RequestMapping("/login")
public class LoginController1
{

	private int width = 120;// 定义图片的width

	private int height = 29;// 定义图片的height

	private int codeCount = 4;// 定义图片上显示验证码的个数

	private int xx = 20;

	private int fontHeight = 18;

	private int codeY = 20;

	char[] codeSequence = { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R',
			'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' };

	private Logger LOGGER = LoggerFactory.getLogger(LoginController1.class);

	@Autowired
	private DocManager docManager;

	@RequestMapping("/login.do")
	@ResponseBody
	public String login(HttpServletRequest request)
	{
		String name = request.getParameter("name");
		String password = request.getParameter("password");
		String code = request.getParameter("code");
		DocBean docBean = null;

		if (StringUtils.isNotBlank(code))
		{
			String oldCode = (String) request.getSession().getAttribute("code");
			if (StringUtils.isNotBlank(oldCode))
			{
				if (!oldCode.equalsIgnoreCase(code))
				{
					return "CODEWARONG";
				}
			}
			else
			{
				return "ABNORMAL";
			}

		}
		else
		{
			return "CODENULL";
		}
		try
		{
			if (StringUtils.isNotBlank(name))
			{
				docBean = docManager.getByName(name);
				if (null != docBean)
				{
					if (StringUtils.isNotBlank(password))
					{
						if (AESUtils.toSHA256(password).equals(docBean.getPassword()))
						{
							if (docBean.getStatus() == 0)
							{
								return "FREEZE";
							}
							else
							{
								request.getSession().setAttribute("currentUser", docBean);
								return Constant.SUCCESS;
							}
						}
					}
				}
			}

		}
		catch (Exception e)
		{
			LOGGER.error("login is error. name=" + name + "&password=" + password + "");
			return "ABNORMAL";
		}
		return Constant.ERROR;
	}

	@RequestMapping(value = "/mainInit.do")
	@ResponseBody
	public Map<String, String> mainInit(HttpServletRequest request)
	{

		Map<String, String> map = new HashMap<String, String>();

		DocBean docBean = (DocBean) request.getSession().getAttribute("currentUser");
		map.put("currentUserName", docBean.getName());
		map.put("currentUserPmsn", docBean.getPmsn());

		return map;
	}

	/**
	 * 退出
	 */
	@RequestMapping("/exit.do")
	public void exit(HttpServletRequest request, HttpServletResponse response)
	{
		request.getSession().invalidate();
		System.out.println("exit!!!");
		try
		{
			response.sendRedirect("/graduation_project/");
		}
		catch (IOException e)
		{
			LOGGER.error("exit  error.", e);
			System.out.println("exit  error." + e);
		}

	}

	/**
	 * 验证码
	 */
	@RequestMapping(value = "/getCode.do")
	public void getCode(HttpServletRequest request, HttpServletResponse response) throws IOException
	{
		// 定义图像buffer
		BufferedImage buffImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
		Graphics gd = buffImg.getGraphics();
		// 创建一个随机数生成器类
		Random random = new Random();
		// 将图像填充为白色
		gd.setColor(Color.WHITE);
		gd.fillRect(0, 0, width, height);
		// 创建字体，字体的大小应该根据图片的高度来定。
		Font font = new Font("Fixedsys", Font.BOLD, fontHeight);
		// 设置字体。
		gd.setFont(font);
		// 画边框。
		gd.setColor(Color.BLACK);
		gd.drawRect(0, 0, width - 1, height - 1);
		// 随机产生40条干扰线，使图象中的认证码不易被其它程序探测到。
		gd.setColor(Color.BLACK);
		for (int i = 0; i < 40; i++)
		{
			int x = random.nextInt(width);
			int y = random.nextInt(height);
			int xl = random.nextInt(12);
			int yl = random.nextInt(12);
			gd.drawLine(x, y, x + xl, y + yl);
		}

		// randomCode用于保存随机产生的验证码，以便用户登录后进行验证。
		StringBuffer randomCode = new StringBuffer();

		int red = 0, green = 0, blue = 0;
		// 随机产生codeCount数字的验证码。
		for (int i = 0; i < codeCount; i++)
		{
			// 得到随机产生的验证码数字。
			String code = String.valueOf(codeSequence[random.nextInt(codeSequence.length - 1)]);
			// 产生随机的颜色分量来构造颜色值，这样输出的每位数字的颜色值都将不同。
			red = random.nextInt(255);
			green = random.nextInt(255);
			blue = random.nextInt(255);
			// 用随机产生的颜色将验证码绘制到图像中。
			gd.setColor(new Color(red, green, blue));
			gd.drawString(code, (i + 1) * xx, codeY);
			// 将产生的四个随机数组合在一起。
			randomCode.append(code);
		}

		// 将四位数字的验证码保存到Session中。
		HttpSession session = request.getSession();
		session.setAttribute("code", randomCode.toString());
		// 禁止图像缓存。
		response.setHeader("Pragma", "no-cache");
		response.setHeader("Cache-Control", "no-cache");
		response.setDateHeader("Expires", 0);
		response.setContentType("image/jpeg");
		// 将图像输出到Servlet输出流中。
		ServletOutputStream out = response.getOutputStream();
		ImageIO.write(buffImg, "jpeg", out);
		out.close();
	}
}
